package thread;

import java.util.PriorityQueue;

public class MyTimerTask implements Comparable<MyTimerTask>{

    //time 是一个ms级别的时间戳
    private long time;
    //时机任务要执行的代码
    private Runnable runnable;

    public long getTime() {
        return time;
    }

    public MyTimerTask(Runnable runnable, long delay) {
        this.runnable = runnable;
        this.time = System.currentTimeMillis() + delay;
    }

    public void run() {
        runnable.run();
    }

    @Override
    public int compareTo(MyTimerTask o) {
        return (int)(this.time - o.time);
    }
}

//通过一个类表示一个任务的进程
class MyTimer {
    //负责扫描任务队列, 执行任务的线程
    private Thread t = null;
    //任务队列
    private PriorityQueue<MyTimerTask> queue = new PriorityQueue<>();

    //锁对象
    private Object locker = new Object();

    public void schedule(Runnable runnable, long delay) {
        synchronized (locker) {
            MyTimerTask task = new MyTimerTask(runnable, delay);
            queue.offer(task);
            //添加新的元素后, 就可以唤醒扫描线程的wait了
            locker.notify();
        }
    }

    //构造方法, 创建扫描线程, 让扫描线程来完成判定和执行
    public MyTimer() {
        t = new Thread(()->{
            //扫描线程循环反复的扫描队首的元素, 然后判定队首元素是不是时间到了
            //如果时间没到, 啥也不干
            //如果时间到了, 就执行这个任务, 并且把这个任务从任务队列中删除
            while (true) {
                try {
                    while (queue.isEmpty()) {
                        locker.wait();
                    }
                    MyTimerTask task = queue.peek();
                    long curTime = System.currentTimeMillis();
                    if(curTime >= task.getTime()) {
                        //当前时间以及达到了任务的执行时间, 可以执行任务了
                        queue.poll();
                        task.run();
                    }else {
                        //当前时间还没到, 暂时先不执行
                        //不能使用 sleep, 会错过新的任务, 也无法释放锁
                        locker.wait(task.getTime()-curTime);
                    }
                }catch (InterruptedException e) {
                  e.printStackTrace();
                }
            }
        });
        t.start();
    }
}